package com.linran.structure_algorithm.数据结构.lesson7_hash;

import lombok.Getter;
import lombok.Setter;

import java.util.Scanner;

public class hashTableDemo {

    public static void main(String[] args) {
        // 创建哈希表
        HashTab hashTab = new HashTab(7);

        // 写一个简单的菜单
        String key = "";
        Scanner scanner = new Scanner(System.in);
        while (true) {
            System.out.println("add: 添加雇员");
            System.out.println("list: 显示雇员");
            System.out.println("find: 查询雇员");
            System.out.println("delete: 删除雇员");
            System.out.println("exit: 退出系统");

            key = scanner.next();
            switch (key) {
                case "add":
                    System.out.println("输入id");
                    int id = scanner.nextInt();
                    System.out.println("输入名字");
                    String name = scanner.next();
                    Emp emp = new Emp(id, name);
                    hashTab.add(emp);
                    break;
                case "list":
                    hashTab.list();
                    break;
                case "find":
                    System.out.println("输入id");
                    int findId = scanner.nextInt();
                    hashTab.findEmpById(findId);
                    break;
                case "delete":
                    System.out.println("输入id");
                    int deleteId = scanner.nextInt();
                    hashTab.delete(deleteId);
                    break;
                case "exit":
                    scanner.close();
                    System.exit(0);
                    break;
                default:
                    break;
            }
        }
    }

}

/**
 * 一个雇员
 */
@Setter
@Getter
class Emp {
    private int id;
    private String name;
    private Emp next;

    public Emp(int id, String name) {
        this.id = id;
        this.name = name;
    }
}

// 创建HashTab 管理多条链表
class HashTab {
    private final int size;
    private final EmpLinkedList[] empLinkedListArray;

    // 构造器
    public HashTab(int size) {
        this.size = size;
        // 初始化链表
        this.empLinkedListArray = new EmpLinkedList[size];
        // 记得链表的初始化
        for (int i = 0; i < this.empLinkedListArray.length; i++) {
            this.empLinkedListArray[i] = new EmpLinkedList();
        }
    }

    // 添加雇员
    public void add(Emp emp) {
        // 根据员工的id，得到该员工对应的链表index
        int empLinkedListNo = hashFun(emp.getId());
        // 将emp添加到对应的链表中
        this.empLinkedListArray[empLinkedListNo].add(emp);
    }

    // 散列函数，使用一个简单的取模方法
    public int hashFun(int id) {
        return id % size;
    }

    // 遍历所有的链表，遍历hashTab
    public void list() {
        for (int i = 0; i < size; i++) {
            empLinkedListArray[i].list(i);
        }
    }

    // 遍历所有链表通过id查找对应的emp
    public void findEmpById(int id) {
        int hashNo = hashFun(id);
        Emp emp = empLinkedListArray[hashNo].findEmpById(id);
        if (emp != null) {
            System.out.printf("找到对应雇员 链表:%d id:%d name:%s %n", hashNo, emp.getId(), emp.getName());
            return;
        }
        System.out.println("未找到对应雇员");
    }

    // 删除操作
    public void delete(int id) {
        int hashNo = hashFun(id);
        empLinkedListArray[hashNo].delete(id);
    }
}

/**
 * 创建EmpLinkedList表示链接
 */
class EmpLinkedList {
    // 头指针，执行第一个Emp,因此我们这个链表的head是直接指向第一个Emp
    private Emp head;// 默认null

    // 添加雇员到链表
    // 说明
    // 1.假定，当添加雇员时,id是自增长，即id的分配总是从小到大
    // 因此我们将雇员直接计入到本链表的最后即可
    public void add(Emp emp) {
        //如果是添加第一个雇员
        if (head == null) {
            head = emp;
            return;
        }
        // 如果不是第一个雇员，则使用一个辅助的指针，帮助定位到最后
        Emp curEmp = head;
        while (true) {
            if (curEmp.getNext() == null) {// 说明到链表最后
                break;
            }
            curEmp = curEmp.getNext(); //后移
        }
        //退出时直接将emp加入链表
        curEmp.setNext(emp);
    }

    //遍历链表的雇员信息
    public void list(int no) {
        // 当前空链表
        if (head == null) {
            System.out.printf("当前链表%d为空%n", no);
            return;
        }
        System.out.printf("当前链表%d的信息为%n", no);
        Emp curEmp = head; //辅助指针
        while (true) {
            System.out.printf("链表:%d id:%d name:%s %n", no, curEmp.getId(), curEmp.getName());
            // 说明到达尾节点
            if (curEmp.getNext() == null) {
                break;
            }
            // 节点后移
            curEmp = curEmp.getNext();
        }
    }

    public Emp findEmpById(int id) {
        //辅助指针
        Emp curEmp = head;
        if (head == null) {
            System.out.println("链表为空");
            return null;
        }
        //找到
        while (curEmp.getId() != id) {
            // 退出
            if (curEmp.getNext() == null) {
                curEmp = null;
                break;
            }
            curEmp = curEmp.getNext(); // 后移动
        }
        return curEmp;
    }

    public void delete(int id) {
        //辅助指针
        Emp pre = head;
        Emp curEmp = head;
        if (head == null) {
            System.out.println("链表为空");
            return;
        }
        //找到
        while (curEmp.getId() != id) {
            // 退出
            if (curEmp.getNext() == null) {
                return;
            }
            pre = curEmp; // 记录前节点
            curEmp = curEmp.getNext(); // 后移动
        }
        // 找到后退出
        if (curEmp.getId() == id) {
            if (curEmp != head) {
                pre.setNext(curEmp.getNext());
            } else {
                head = curEmp.getNext();
            }
            System.out.println("删除成功");
        }
    }
}